import os
import cv2

# 要被合成的多张图片所在文件夹
# 路径分隔符最好使用“/”,而不是“\”,“\”本身有转义的意思；或者“\\”也可以。
# 因为是文件夹，所以最后还要有一个“/”
file_path = 'D:/code/Fast-Lane-Detection/Tusimple/list/my_test.txt'

with open(file_path) as f:
    text = f.readlines()

# VideoWriter是cv2库提供的视频保存方法，将合成的视频保存到该路径中
# 'MJPG'意思是支持jpg格式图片
# fps = 5代表视频的帧频为5，如果图片不多，帧频最好设置的小一点
# (1280,720)是生成的视频像素1280*720，一般要与所使用的图片像素大小一致，否则生成的视频无法播放
# 定义保存视频目录名称和压缩格式，像素为1280*720
video = cv2.VideoWriter('D:/code/Fast-Lane-Detection/test.mp4', cv2.VideoWriter_fourcc('m', 'p', '4', 'v'),8, (1280, 720) )

for i, line in enumerate(text):
    img_path ='D:/code/Fast-Lane-Detection/Tusimple/' + line.split()[0]
    print(img_path)
    img = cv2.imread(img_path)
    cv2.imshow("1", img)
    # 写入视频
    video.write(img)
    cv2.waitKey(100)
# 释放资源
video.release()






